from langchain_community.chat_models import ChatTongyi
from langchain_core.prompts.prompt import PromptTemplate
from langchain.chains import LLMChain
from langchain.chains import ConversationChain

import os

os.environ["DASHSCOPE_API_KEY"] = "sk-9d8f1914800e497f8717144e860f99bc"
# 调用模型
llm = ChatTongyi()
default_chain = ConversationChain(llm=llm, output_key="AI")
# 第一轮对话
print(default_chain.invoke("你好"))
# 第二轮对话
print(default_chain.invoke("你是谁"))